package com.eian.boot.security.utils;

import cn.dev33.satoken.session.SaSession;
import cn.dev33.satoken.stp.StpUtil;
import cn.dev33.satoken.stp.parameter.SaLoginParameter;
import com.eian.boot.common.core.model.model.LoginUser;
import com.eian.boot.security.handler.LogoutHandler;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.ObjectUtils;
import org.apache.commons.lang3.StringUtils;

import java.util.Optional;


/**
 * 登录鉴权助手
 */
@Slf4j
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class SessionHelper {
    public static final String LOGIN_USER_KEY = "login-user";

    /**
     * 登录
     *
     * @param loginUser 登录用户信息
     * @param model     配置参数
     * @return 鉴权Token
     */
    public static String login(LoginUser loginUser, SaLoginParameter model) {
        model = ObjectUtils.isNotEmpty(model) ? model : new SaLoginParameter();
        StpUtil.login(loginUser.getId(), model);
        StpUtil.getTokenSession().set(LOGIN_USER_KEY, loginUser);
        return getToken();
    }

    /**
     * 登录
     *
     * @param loginUser 登录用户信息
     * @return 鉴权Token
     */
    public static String login(LoginUser loginUser) {
        return login(loginUser, null);
    }

    /**
     * 获取用户(多级缓存)
     */
    public static <T extends LoginUser> T getLoginUser() {
        return getLoginUser(null);
    }

    /**
     * 获取用户基于token
     */
    @SuppressWarnings("unchecked cast")
    public static <T extends LoginUser> T getLoginUser(String token) {
        SaSession session;
        if (StringUtils.isBlank(token)) {
            session = StpUtil.getTokenSession();
        } else {
            session = StpUtil.getTokenSessionByToken(token);
        }

        if (ObjectUtils.isEmpty(session)) {
            return null;
        }
        return (T) Optional.ofNullable(session.get(LOGIN_USER_KEY)).orElseThrow(() -> new RuntimeException("未获取到有效用户信息"));
    }

    /**
     * 获取用户id
     */
    public static Long getUserId() {
        return StpUtil.getLoginIdAsLong();
    }

    /**
     * 获取用户id
     */
    public static String getUserIdStr() {
        return StpUtil.getLoginIdAsString();
    }

    /**
     * 获取用户账户
     */
    public static String getUsername() {
        return getLoginUser().getUsername();
    }

    /**
     * 检查当前用户是否已登录
     *
     * @return 结果
     */
    public static boolean isLogin() {
        try {
            return ObjectUtils.isNotEmpty(getLoginUser());
        } catch (Exception e) {
            return false;
        }
    }

    public static String getToken() {
        return StpUtil.getTokenValue();
    }

    /**
     * @author alex.meng
     * @description 退出登录
     **/
    public static void logout() {
        StpUtil.logout();
    }

    /**
     * @author alex.meng
     * @description 带后续处理
     **/
    public static void logout(LogoutHandler... handlers) {
        String loginId = StpUtil.getLoginIdAsString();  // 退出前获取信息
        String tokenValue = StpUtil.getTokenValue();
        // 执行退出
        StpUtil.logout();
        // 执行后续处理
        if (handlers != null) {
            for (LogoutHandler handler : handlers) {
                try {
                    handler.afterLogout(loginId, tokenValue);
                } catch (Exception e) {
                    log.error("退出后处理失败: {}", e.getMessage(), e);
                }
            }
        }
    }

    /**
     * @author alex.meng
     * @description 根据Id踢人下线
     **/
    public static void logout(Long loginId) {
        StpUtil.logout(loginId);
    }
}
